from core.agents.plan.agents import build_plan_team_graph

plan_team = build_plan_team_graph()


if __name__ == '__main__':
    # res = plan_team.invoke({"messages": "what's (3 + 5) x 12?"})
    res = plan_team.invoke({"messages": "泡制一壶茉莉花茶需要分为几个步骤?"})
    print(res['messages'][-1].content)

